Numpy getting indices of subarrays of a 1darray that are of a specific size with numpy.where()